"""
main.py 系统主界面
"""

def main_menu():
    """
    显示系统主菜单，并获取用户合法输入
    """
    print('-------------------------')
    print('1.查询格式化消息')
    print('2.添加格式化消息')
    print('3.修改格式化消息')
    print('4.删除格式化消息')
    print('5.退出')

def get_user_choice():
    """
    获取用户输入的选项
    """
    try:
        c = int(input('请选择（1-5）：'))
        if 1 <= c <= 5:
            return c
        else:
            # 输入非法，将显示帮助信息
            return 0
    except ValueError:
        return 0
    
def add_formatted_message():
    print('添加格式化消息功能')

def get_formatted_message():
    print('查询格式化消息功能')

def update_formatted_message():
    print('修改格式化消息功能')

def del_formatted_message():
    print('删除格式化消息功能')

def call_function(choice=0):
    """
    调用指定的函数
    """
    if choice == 0:
        print('请选择正确的选项！')
    elif choice == 1: # 查询格式化消息
        get_formatted_message()
    elif choice == 2: # 添加格式化消息
        add_formatted_message()
    elif choice == 3: # 修改格式化消息
        update_formatted_message()
    elif choice == 4: # 删除格式化消息
        del_formatted_message()
    elif choice == 5: # 退出系统
        print('退出系统')
        exit(0)

def run():
    """
    系统入口函数，显示主菜单，并执行相应操作
    """
    while True:
        main_menu()
        t = get_user_choice()
        call_function(t)

run()